import json
from bson import ObjectId


class JSONEncoder(json.JSONEncoder):
    def default(self, data):
        if isinstance(data, ObjectId):
            return str(data)
        return json.JSONEncoder.default(self, data)
